import request from '@/config/axios'

// 数据集 VO
export interface DatasetVO {
  id: number // promptID
  name: string // 数据集名称
  promptIds: string // promptId集合，逗号隔开
  indicatorCodes: string | any // indicatorCode集合，逗号隔开
  status?: number | undefined // 状态（0正常 1停用）
  remark: string // 数据集描述
  datasetTypeId: number // 数据集类型id
  num?: number | undefined // 数据量
}

// 数据集 API
export const DatasetApi = {
  // 查询数据集分页
  getDatasetPage: async (params: any) => {
    return await request.get({ url: `/multiModel/dataset/page`, params })
  },

  // 查询数据集详情
  getDataset: async (id: number) => {
    return await request.get({ url: `/multiModel/dataset/get?id=` + id })
  },

  // 新增数据集
  createDataset: async (formData) => {
    return await request.upload({ url: `/multiModel/dataset/create`, data: formData })
  },

  // 修改数据集
  updateDataset: async (data: DatasetVO) => {
    return await request.put({ url: `/multiModel/dataset/update`, data })
  },

  // 删除数据集
  deleteDataset: async (id: number) => {
    return await request.delete({ url: `/multiModel/dataset/delete?id=` + id })
  },

  // 导出数据集 Excel
  exportDataset: async (params) => {
    return await request.download({ url: `/multiModel/dataset/export-excel`, params })
  },

  importDatasetTemplate: async () => {
    return await request.download({url: `multiModel/dataset/get-import-template`})
  }
}
